package org.example;

import ws.schild.jave.*;

import java.io.File;
import java.math.BigDecimal;
import java.util.List;

/**
 * @ClassName videoTest
 * @Description
 * @Author zhangsw
 * @Date 2020/10/19 8:43
 * @Version 1.0
 */
public class videoTest {
    public static void main(String[] args) {
//        double a = 1024*25*1024;//26143434
//        System.err.println(a);
//        Integer a = 100;
//        Integer b = 101;
//        if (a.equals(100)) {
//            System.err.println("a等于100");
//        }
//        if (!b.equals(100)) {
//            System.err.println("b不等于100");
//        }


//        File source = new File("D:\\qmt\\imgPath\\2020-10-18\\d976db7f-6d4e-4a19-8343-b7b1e2396e6b.mp4");
//        File target = new File("D:\\qmt\\imgPath\\8343-b7b1e2396e6b.mp4");
//        try {
//            // 音频编码设置
//            AudioAttributes audio = new AudioAttributes();
//            audio.setCodec("libmp3lame");
//            audio.setBitRate(new Integer(64000));
//            audio.setChannels(new Integer(1));
//            audio.setSamplingRate(new Integer(22050));
//
//            // 视频编码设置
//            VideoAttributes video = new VideoAttributes();
//            video.setCodec("flv");
//            video.setBitRate(new Integer(160000));
//            video.setFrameRate(new Integer(15));
//            video.setSize(new VideoSize(400, 300));
//
//            // 视频转码编码设置
//            EncodingAttributes attrs = new EncodingAttributes();
//            attrs.setFormat("flv");
//            attrs.setAudioAttributes(audio);
//            attrs.setVideoAttributes(video);
//
//            // 编码器
//            Encoder encoder = new Encoder();
//            encoder.encode(source, target, attrs);
//
//            System.out.println("压缩完成...");
//        } catch (EncoderException e) {
//            e.printStackTrace();
//
//        }

//        String savePath = "/tmp/tempVideo/";
//        String savePath = "D:\\qmt\\imgPath\\2020-10-18\\e6842c79-b36d-4d87-8d22-fba08b87da08.mp4";
//        File saveDir = new File(savePath);
//        if(!saveDir.exists()){
//            saveDir.mkdirs();
//        }
//        File file = compressionVideo(saveDir, "e6842c79-b36d-4d87-8d22-fba08b87da082.mp4");
//
//        System.out.println(file.getName());
        String savePath = "D:\\qmt\\imgPath\\2020-10-19\\4c3bc843-3044-4c4c-bb28-5a565778f803.mp4";
        File saveDir = new File(savePath);
        if(!saveDir.exists()){
            saveDir.mkdirs();
        }
        System.out.println(saveDir.isFile());
        System.out.println(saveDir.getName());
        File file = compressionVideo(saveDir, "D:\\qmt\\imgPath\\2020-10-19\\qqq.mp4");
        System.out.println(file.getName());

    }

    /**
     * 传视频File对象，返回压缩后File对象信息
     *
     * @param source
     */
    public static File compressionVideo(File source, String newPath) {
        if (source == null) {
            return null;
        }
//        String absolutePath = source.getAbsolutePath();
//        System.err.println("absolutePath="+absolutePath);
//        String newPath = source.getAbsolutePath().substring(0, source.getAbsolutePath().lastIndexOf("\\")).concat(picName);
//        System.out.println("newPath="+newPath);




        File target = new File(newPath);
        try {
            MultimediaObject object = new MultimediaObject(source);
            AudioInfo audioInfo = object.getInfo().getAudio();
            // 根据视频大小来判断是否需要进行压缩,
            int maxSize = 100;
            double mb = Math.ceil(source.length() / 1048576);
            int second = (int) object.getInfo().getDuration() / 1000;
            BigDecimal bd = new BigDecimal(String.format("%.4f", mb / second));
            System.out.println("开始压缩视频了--> 视频每秒平均 " + bd + " MB ");
            // 视频 > 100MB, 或者每秒 > 0.5 MB 才做压缩， 不需要的话可以把判断去掉
            boolean temp = mb > maxSize || bd.compareTo(new BigDecimal(0.5)) > 0;
            if (temp) {
                long time = System.currentTimeMillis();
                //TODO 视频属性设置
                int maxBitRate = 128000;
                int maxSamplingRate = 44100;
                int bitRate = 800000;
                int maxFrameRate = 20;
                int maxWidth = 1280;

                AudioAttributes audio = new AudioAttributes();
                // 设置通用编码格式
                audio.setCodec("aac");
                // 设置最大值：比特率越高，清晰度/音质越好
                // 设置音频比特率,单位:b (比特率越高，清晰度/音质越好，当然文件也就越大 128000 = 182kb)
                if (audioInfo.getBitRate() > maxBitRate) {
                    audio.setBitRate(new Integer(maxBitRate));
                }

                // 设置重新编码的音频流中使用的声道数（1 =单声道，2 = 双声道（立体声））。如果未设置任何声道值，则编码器将选择默认值 0。
                audio.setChannels(audioInfo.getChannels());
                // 采样率越高声音的还原度越好，文件越大
                // 设置音频采样率，单位：赫兹 hz
                // 设置编码时候的音量值，未设置为0,如果256，则音量值不会改变
                // audio.setVolume(256);
                if (audioInfo.getSamplingRate() > maxSamplingRate) {
                    audio.setSamplingRate(maxSamplingRate);
                }

                //TODO 视频编码属性配置
                VideoInfo videoInfo = object.getInfo().getVideo();
                VideoAttributes video = new VideoAttributes();
                video.setCodec("h264");
                //设置音频比特率,单位:b (比特率越高，清晰度/音质越好，当然文件也就越大 800000 = 800kb)
                if (videoInfo.getBitRate() > bitRate) {
                    video.setBitRate(bitRate);
                }

                // 视频帧率：15 f / s  帧率越低，效果越差
                // 设置视频帧率（帧率越低，视频会出现断层，越高让人感觉越连续），视频帧率（Frame rate）是用于测量显示帧数的量度。所谓的测量单位为每秒显示帧数(Frames per Second，简：FPS）或“赫兹”（Hz）。
                if (videoInfo.getFrameRate() > maxFrameRate) {
                    video.setFrameRate(maxFrameRate);
                }

                // 限制视频宽高
                int width = videoInfo.getSize().getWidth();
                int height = videoInfo.getSize().getHeight();
                if (width > maxWidth) {
                    float rat = (float) width / maxWidth;
                    video.setSize(new VideoSize(maxWidth, (int) (height / rat)));
                }

                EncodingAttributes attr = new EncodingAttributes();
                attr.setFormat("mp4");
                attr.setAudioAttributes(audio);
                attr.setVideoAttributes(video);

                // 速度最快的压缩方式， 压缩速度 从快到慢： ultrafast, superfast, veryfast, faster, fast, medium,  slow, slower, veryslow and placebo.
//                attr.setPreset(PresetUtil.VERYFAST);
//                attr.setCrf(27);
//                // 设置线程数
//                attr.setEncodingThreads(Runtime.getRuntime().availableProcessors()/2);

                Encoder encoder = new Encoder();
                encoder.encode(new MultimediaObject(source), target, attr);
                System.out.println("压缩总耗时：" + (System.currentTimeMillis() - time) / 1000);
                return target;
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (target.length() > 0) {
                source.delete();
            }
        }
        return source;
    }
}
